package dataConverter;

import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.Date;
import java.util.HashMap;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

public class Xml2JSON {
	
	/**
	 * 将xml转换为JSON对象
	 * 
	 * @param xml xml字符串
	 * @return
	 * @throws Exception
	 */
	public static JSONObject xmltoJson(String xml) throws Exception {
		JSONObject jsonObject = new JSONObject();
		Document document = DocumentHelper.parseText(xml);
		// 获取根节点元素对象
		Element root = document.getRootElement();
		iterateNodes(root, jsonObject);
		return jsonObject;
	}

	public static JSONObject xmlToJSOM(String XMLStr) throws DocumentException {
		JSONObject jo = new JSONObject();

		Document docment = DocumentHelper.parseText(XMLStr);
		Element element = docment.getRootElement();
		iterateNodes(element, jo);
		return jo;
	}

	/**
	 * 递归遍历元素，转为xml节点形式
	 * 
	 * @param node 元素
	 * @param json 将元素遍历完成之后放的JSON对象
	 */
	public static void iterateNodes(Element node, JSONObject json) {
		// 获取当前元素的名称
		String nodeName = node.getName();
//		if(nodeName.equals("contact")) {
//			String key = node.attributeValue("id");
//			String value = node.attributeValue("name");
//			System.out.println("key =" + key + ", value ="+ value);
//		}
		// 判断已遍历的JSON中是否已经有了该元素的名称
		if (json.containsKey(nodeName)) {
			// 该元素在同级下有多个
			Object Object = json.get(nodeName);
			JSONArray array = null;
			if (Object instanceof JSONArray) {
				array = (JSONArray) Object;
			} else {
				array = new JSONArray();
				array.add(Object);
			}
			// 获取该元素下所有子元素
			List<Element> listElement = node.elements();
			if (listElement.isEmpty()) {
				// 该元素无子元素，获取元素的值
				String nodeValue = node.getTextTrim();
				array.add(nodeValue);
				json.put(nodeName, array);
				return;
			}
			// 有子元素
			JSONObject newJson = new JSONObject();
			// 遍历所有子元素
			for (Element e : listElement) {
				// 递归
				iterateNodes(e, newJson);
			}
			array.add(newJson);
			json.put(nodeName, array);
			return;
		}
		// 该元素同级下第一次遍历
		// 获取该元素下所有子元素
		List<Element> listElement = node.elements();
		if (listElement.isEmpty()) {
			// 该元素无子元素，获取元素的值
			String nodeValue = node.getTextTrim();
			json.put(nodeName, nodeValue);
			return;
		}
		// 有子节点，新建一个JSONObject来存储该节点下子节点的值
		JSONObject object = new JSONObject();
		// 遍历所有一级子节点
		for (Element e : listElement) {
			// 递归
			iterateNodes(e, object);
		}
		json.put(nodeName, object);
		return;
	}

	public static void main(String[] args) throws Exception {
		String xml = "<?xml version=\"1.0\" encoding=\"gbk\"?>\r\n"
				+ "<contact-list>\r\n"
				+ "<aaa>\r\n"
				+ "	<contact id=\"001\" name=\"eric1\"/>\r\n"
				+ "	<contact id=\"002\" name=\"eric2\" />\r\n"
				+ "	<contact id=\"003\" name=\"eric3\" /></aaa>\r\n"
				+ "</contact-list>";
		JSONObject jsonObject = xmlToJSOM(xml);
		System.out.println(jsonObject);
		
//		JSONObject jo = jsonObject.getJSONObject("Response");
//		JSONArray jsonArr = jo.getJSONArray("Item");
//		JSONObject tempJsonObject = new JSONObject();
//		int index = 0;
//		for (int i = 0; i < jsonArr.size(); i++) {
//			JSONObject joi = jsonArr.getJSONObject(i);
//			int schemaId = joi.getIntValue("SchemaId");
//			for (int j = 0; j < jsonArr.size(); j++) {
//				JSONObject joj = jsonArr.getJSONObject(j);
//				int id = joi.getIntValue("SchemaId");
//				if (schemaId > id) {
//					tempJsonObject = joi;
//					joi = joj;
//					joj = tempJsonObject;
//				}
//				index++;
//				if (index == 10000) {
//					System.out.println(111);
//					System.out.println(joi.toString());
//					System.out.println(joj.toString());
//					System.out.println(tempJsonObject.toString());
//				}
//			}
//		}
//
//		System.out.println("start....");
//		createJsonFile(jo.toString(), "./src/main/resources", "jsonArr");
//		createJsonFile(jsonObject.toString(), "./src/main/resources", "111");
	}

	public static boolean createJsonFile(String jsonString, String filePath, String fileName) {
		// 标记文件生成是否成功
		boolean flag = true;
		// 拼接文件完整路径
		String fullPath = filePath + File.separator + fileName + ".json";

		// 生成json格式文件
		try {
			// 保证创建一个新文件
			File file = new File(fullPath);
			if (!file.getParentFile().exists()) { // 如果父目录不存在，创建父目录
				file.getParentFile().mkdirs();
			}
			if (file.exists()) { // 如果已存在,删除旧文件
				file.delete();
			}
			file.createNewFile();

			if (jsonString.indexOf("'") != -1) {
				// 将单引号转义一下，因为JSON串中的字符串类型可以单引号引起来的
				jsonString = jsonString.replaceAll("'", "\\'");
			}
			if (jsonString.indexOf("\"") != -1) {
				// 将双引号转义一下，因为JSON串中的字符串类型可以单引号引起来的
				jsonString = jsonString.replaceAll("\"", "\\\"");
			}

			if (jsonString.indexOf("\r\n") != -1) {
				// 将回车换行转换一下，因为JSON串中字符串不能出现显式的回车换行
				jsonString = jsonString.replaceAll("\r\n", "\\u000d\\u000a");
			}
			if (jsonString.indexOf("\n") != -1) {
				// 将换行转换一下，因为JSON串中字符串不能出现显式的换行
				jsonString = jsonString.replaceAll("\n", "\\u000a");
			}

			// 格式化json字符串
//	           jsonString = JSONUtil.formatJsonStr(jsonString);

			// 将格式化后的字符串写入文件
			Writer write = new OutputStreamWriter(new FileOutputStream(file), "UTF-8");
			write.write(jsonString);
			write.flush();
			write.close();
		} catch (Exception e) {
			flag = false;
			e.printStackTrace();
		}
		// 返回是否成功的标记
		return flag;
	}

}
